from pymysql import connect
from config import *


class DB(object):
    # 连接到数据库

    def __init__(self):
        self.conn = connect(host=DB_HOST,
                            port=DB_PORT,
                            database=DB_NAME,
                            user=DB_USER,
                            password=DB_PASS,
                            charset='utf8',

                            )
        # 获取游标
        self.cursor = self.conn.cursor()

    def get_one(self, sql):
        """ 使用sql查询用户信息"""
        # 执行sql语句
        self.cursor.execute(sql)

        # 获取查询到的数据
        query_result = self.cursor.fetchone()
        # 判断是否有结果
        if not query_result:
            return None

        # 获取字段
        fileds = [filed[0] for filed in self.cursor.description]
        # 使用字段和数据合成字典，返回
        return_data = {}
        for filed, value in zip(fileds, query_result):
            return_data[filed] = value
        return return_data


    print('xxxx')


    def close(self):
        """释放数据库资源"""
        self.cursor.close()
        self.conn.close()


